import { google } from '@ai-sdk/google';
import { generateText } from 'ai';
import 'dotenv/config';

async function main() {
  const videoUrl =
    'http://commondatastorage.googleapis.com/gtv-videos-bucket/sample/BigBuckBunny.mp4';
  const resultWithMediaLow = await generateText({
    model: google('gemini-2.5-flash'),
    messages: [
      {
        role: 'user',
        content: [
          {
            type: 'text',
            text: `Summarize the video.`,
          },
          {
            type: 'file',
            data: videoUrl,
            mediaType: 'video/mp4',
          },
        ],
      },
    ],
    providerOptions: {
      google: {
        mediaResolution: 'MEDIA_RESOLUTION_LOW',
      },
    },
  });

  console.log(
    "Token usage with 'MEDIA_RESOLUTION_LOW':",
    resultWithMediaLow.totalUsage.totalTokens,
  );

  const resultWithoutMedia = await generateText({
    model: google('gemini-2.5-flash'),
    messages: [
      {
        role: 'user',
        content: [
          {
            type: 'text',
            text: `Summarize the video.`,
          },
          {
            type: 'file',
            data: videoUrl,
            mediaType: 'video/mp4',
          },
        ],
      },
    ],
  });

  console.log(
    "Token usage without 'mediaResolution':",
    resultWithoutMedia.totalUsage.totalTokens,
  );
}

main().catch(console.error);